from base import LlmCorrectnessTest
from config_reader import Config

def main():
    # 读取测试配置信息
    config_file = './config/test-config.yaml'
    config = Config(config_file)

    # llperf测试参数
    LlmCorrectnessParams = config.get_section_as_dict('LlmCorrectnessParams')

    # 创建模型自动化测试实例
    llm_correctness_test = LlmCorrectnessTest(**LlmCorrectnessParams)
    # 运行测试
    llm_correctness_test.run_model_test()
    # 提取结果到excel文件
    llm_correctness_test.save_test_reulsts()

if __name__ == "__main__":
    main()
